#include "mystdio.h"
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <malloc.h>
#include <assert.h>

MY_FILE* my_fopen(const char* path, const char* mode)
{
    //1.识别标志位
    int flag = 0;
    if(strcmp(mode, "r") == 0) flag |= O_RDONLY;
    else if(strcmp(mode, "w") == 0) flag |= (O_CREAT | O_WRONLY | O_TRUNC);
    else if(strcmp(mode, "a") == 0) flag |= (O_CREAT | O_WRONLY | O_APPEND);
    else{
        //其他方式
    }

    //2.打开文件
    int fd = 0;
    mode_t m = 0666;
    if(flag & O_CREAT) fd = open(path, flag, m);
    else fd = open(path, flag);

    if(fd < 0)
    {
        return NULL;
    }

    //3.为了后面给用户返回MY_FILE对象，要先构建出来
    MY_FILE* mf = (MY_FILE*)malloc(sizeof(MY_FILE));

    if(mf == NULL)
    {
        close(fd);
        return NULL;
    }

    //4.初始化MY_FILE对象
    mf->fd = fd;
    mf->flags = BUFF_LINE;
    memset(mf->outputbuffer, '\0', sizeof(mf->outputbuffer));
    mf->current = 0;

    //5.返回打开的文件
    return mf;
}

//参数：ptr：缓冲区，size：写入的单个字段的大小，nemeb：写几个字段，stream：文件流。
//返回值：C语言中返回的是nemeb写入的字段个数，我们这里返回字节数。
//将缓冲区中的数据写入文件流（实际上是写入了outputbuffer中）
size_t my_fwrite(const void* ptr, size_t size, size_t nemeb, MY_FILE* stream)
{
    //1.如果缓冲区满了，就直接写入
    if(stream->current == NUM) my_fflush(stream);

    //2.根据缓冲区剩余的情况，进行数据拷贝
    size_t user_size = size * nemeb;//要写入的字节数
    size_t my_size = NUM - stream->current;//缓冲区还剩多少字节数
    size_t writen = 0;//实际写入了多少字节数，用作返回值。
    if(my_size >= user_size)
    {
        memcpy(stream->outputbuffer + stream->current, ptr, user_size);
        //3.拷贝完后更新计数器
        stream->current += user_size;
        writen = user_size;
    }
    else
    {
        memcpy(stream->outputbuffer + stream->current, ptr, my_size);
        //3.拷贝完后更新计数器
        stream->current += my_size;
        writen = my_size;
    }

    //4.开始刷新数据
    if(stream->flags & BUFF_ALL)
    {
        if(stream->current == NUM) my_fflush(stream);
    }
    else if(stream->flags & BUFF_LINE)
    {
        if(stream->outputbuffer[stream->current] == '\n')
        {
            my_fflush(stream);
        }
    }
    else
    {

    }
}

int my_fflush(MY_FILE* fp)
{
    assert(fp);
    write(fp->fd, fp->outputbuffer, fp->current);
    fp->current = 0;

    fsync(fp->fd);//将缓冲区数据刷新到磁盘
    return 0;
}

int my_fclose(MY_FILE* fp)
{
    assert(fp);
    //1.刷新缓冲区
    if(fp->current > 0) my_fflush(fp);
    //2.关闭文件
    close(fp->fd);
    //3.释放堆空间（结构体中的缓冲区）
    free(fp);
    //4.指针置空
    fp = NULL;
}